"""
将服务器的发送的数据保存到文件中
@Time ： 2021/11/24 17:19
@Auth ： ChenZhangKun
"""
import socket
from socket import *


def main():
    # 创建套接字
    tcp_socket = socket(AF_INET, SOCK_STREAM)
    # 连接服饰无武器
    tcp_socket.connect(("localhost", 8080))
    # 获取下载的文件名称
    download_file_name = input("请输入文件名称： ")
    # 将文件发到服务器
    tcp_socket.send(download_file_name.encode("utf-8"))
    # 接收数据
    recv_data = tcp_socket.recv(1024)
    # 保存数据 ->将发送的数据保存到文件
    if recv_data:
        with open("[新]" + download_file_name, "wb") as f:
            f.write(recv_data)
    tcp_socket.close()


if __name__ == '__main__':
    main()
